from distutils.log import info
from openpyxl import load_workbook
# 打开相应的excle表格   相对路径  绝对路径 
wb_us = load_workbook(filename="C:/Users/G15/Desktop/stu_info.xlsx")
def stu_menu():
    print("-"*20 + "欢迎来到学生主界面" + "-"*20)
    print("1、查询学生信息")
    print('2、修改学生信息')
    print('3、删除学生信息')
    print('4、退出')
    text = input("请输入相应的值：")
    if text == "1" :
        stu_query()
    elif text == "2":
        stu_class()
    elif text == "3":
        stu_num()
    elif text == "4" :
        stu_add()
    elif text == "5":
        print("退出系统成功。")
        return
    else:
        print("请按要求输入数值。")
def stu_add():
    stu_class = input("请输入班级（如数据3班）：")
    stu_wb = wb_us[stu_class]
    stu_id = input("请输入增加的学号：")
    stu_name = input("请输入增加的姓名：")
    stu_age = input("请输入增加的年龄：")
    stu_sex = input("请输入增加的性别：")
    stu_bir = input("请输入增加的出生日期：")
    stu_adr = input("请输入增加的籍贯：")
    print(f'序号：{stu_id},姓名：{stu_name},年龄：{stu_age}，性别：{stu_sex},出生年月：{stu_bir},籍贯：{stu_adr}')
# 查询学生信息
def stu_query():
    print("-"*20)
    print("1、按照班级查询")
    print("2、按照学号查询")
    num = input("请输入查询的数值：")
    if num == "1":
        stu_class()
    elif num == "2":
        stu_num()
    else:
        print("请输入正确的数值")
    
    # print("查询学生信息")
    # 按照班级查询的方法
def stu_class():
    # 按照班级查询
    stu_class = input("请输入班级(如数据三班)：")
    # 打开子表
    stu_wb = wb_us[stu_class]
    # 遍历相应的表格 把数据显示出来
    for i in range(2,stu_wb.max_row+1):
        print(f'序号：{stu_wb[i][0].value},姓名：{stu_wb[i][1].value},年龄：{stu_wb[i][2].value}，性别：{stu_wb[i][3].value},出生年月：{stu_wb[i][4].value},籍贯：{stu_wb[i][5].value},联系方式：{stu_wb[i][6].value}')
    # print("修改学生信息")
def stu_num():
    # 先找班级
    stu_class = input("请输入班级(如数据3班)：")
    # 打开子表
    stu_wb = wb_us[stu_class]
    # 匹配学号
    stu_id = int(input("请输入学号："))
    # 从表格中找到相应的学号
    for i in range(2,stu_wb.max_row+1):
        # 判断输入的学号
        if stu_id == stu_wb[i][0].value:
            print(f"序号：{stu_wb[i][0].value},姓名：{stu_wb[i][1].value},年龄：{stu_wb[i][2].value}，性别：{stu_wb[i][3].value},出生年月：{stu_wb[i][4].value},籍贯：{stu_wb[i][5].value},联系方式：{stu_wb[i][6].value}")
def stu_change():
    stu_class = input("请输入班级(如数据3班)：")
    # 打开子表
    stu_wb = wb_us[stu_class]
    # 匹配学号
    stu_id = int(input("请输入学号："))
    # 从表格中找到相应的学号
    for i in range(2,stu_wb.max_row+1):
        # 判断输入的学号
        if stu_id == stu_wb[i][0].value:
            print(f"序号：{stu_wb[i][0].value},姓名：{stu_wb[i][1].value},年龄：{stu_wb[i][2].value}，性别：{stu_wb[i][3].value},出生年月：{stu_wb[i][4].value},籍贯：{stu_wb[i][5].value},联系方式：{stu_wb[i][6].value}")
            # 修改相应的内容
            info = input ("是否修改1、修改 2、退出")
            if info == "1":
                stu_name = input("请输入修改的姓名：")
                stu_age = input("请输入修改的年龄：")
                stu_sex = input("请输入修改的性别：")
                stu_bir = input("请输入修改的出生日期：")
                stu_adr = input("请输入修改的籍贯：")
                # 修改数据 --赋值
                stu_wb[i][1].value = stu_name
                stu_wb[i][2].value = stu_age
                stu_wb[i][3].value = stu_sex
                stu_wb[i][4].value = stu_bir
                stu_wb[i][5].value = stu_adr
                wb_us.save(filename="C:/Users/G15/Desktop/stu_info.xlsx")
                print("修改成功！")
            else:
                return
def stu_delete():
        stu_class = input("请输入班级(如数据4班)：")
        # 打开子表
        stu_wb = wb_us[stu_class]
    # 匹配学号
        stu_id = int(input("请输入学号："))
    # 从表格中找到相应的学号
        for i in range(2,stu_wb.max_row+1):
        # 判断输入的学号
            if stu_id == stu_wb[i][0].value:
                print(f"序号：{stu_wb[i][0].value},姓名：{stu_wb[i][1].value},年龄：{stu_wb[i][2].value}，性别：{stu_wb[i][3].value},出生年月：{stu_wb[i][4].value},籍贯：{stu_wb[i][5].value},联系方式：{stu_wb[i][6].value}")
                # 修改相应的内容
                info =input("是否删除：1、删除 2、退出")
                if info == "1":
                    stu_wb.delete_rows(i)
                    wb_us.save(filename="C:/Users/G15/Desktop/stu_info.xlsx")
                print("删除学生信息成功")
            else:
                print("删除学生信息失败")
